package com.xc.controller;


import cn.hutool.core.io.FileUtil;
import cn.hutool.core.io.resource.ResourceUtil;
import com.itextpdf.text.DocumentException;
import com.itextpdf.text.pdf.BaseFont;
import lombok.extern.slf4j.Slf4j;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.ResponseBody;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.multipart.MultipartFile;
import org.xhtmlrenderer.pdf.ITextFontResolver;
import org.xhtmlrenderer.pdf.ITextRenderer;

import javax.servlet.http.HttpServletResponse;
import java.io.*;
import java.net.URLEncoder;
import java.util.Objects;
import java.util.UUID;

/**
 * 上传下载测试接口
 */
@RestController  //相当于@Controller @responseBody
@RequestMapping
@ResponseBody
@Slf4j
public class UpDownController {
    @RequestMapping("/upload")
    public Object upload(@RequestParam MultipartFile file) {
        if (!file.isEmpty()) {
            String uploadPath = "D:\\aaa";
            // 如果目录不存在则创建
            File uploadDir = new File(uploadPath);
            if (!uploadDir.exists()) {
                uploadDir.mkdir();
            }
            String OriginalFilename = file.getOriginalFilename();//获取原文件名
            String suffixName = Objects.requireNonNull(OriginalFilename).substring(OriginalFilename.lastIndexOf("."));//获取文件后缀名
            //重新随机生成名字
            String filename = UUID.randomUUID() + suffixName;
            File localFile = new File(uploadPath + "\\" + filename);
            try {
                file.transferTo(localFile); //把上传的文件保存至本地
                 //这里应该把filename保存到数据库,供前端访问时使用
                return filename;//上传成功，返回保存的文件地址
            } catch (IOException e) {
                log.error("上传失败:[{}]",e.toString());
                return "";
            }
        } else {
            log.error("文件为空");
            return "";
        }
    }

    @RequestMapping("/download")
    public String download(HttpServletResponse response) {
        try {
            // path是指想要下载的文件的路径
            File file = new File("D:\\aaa\\testoone3.pdf");
            log.info(file.getPath());
            // 获取文件名
            String filename = file.getName();
            // 获取文件后缀名
            String ext = filename.substring(filename.lastIndexOf(".") + 1).toLowerCase();
            log.info("文件后缀名：" + ext);

            // 将文件写入输入流
            FileInputStream fileInputStream = new FileInputStream(file);
            InputStream fis = new BufferedInputStream(fileInputStream);
            byte[] buffer = new byte[fis.available()];
            fis.read(buffer);
            fis.close();

            // 清空response
            response.reset();
            // 设置response的Header
            response.setCharacterEncoding("UTF-8");
            //Content-Disposition的作用：告知浏览器以何种方式显示响应返回的文件，用浏览器打开还是以附件的形式下载到本地保存
            //attachment表示以附件方式下载   inline表示在线打开   "Content-Disposition: inline; filename=文件名.mp3"
            // filename表示文件的默认名称，因为网络传输只支持URL编码的相关支付，因此需要将文件名URL编码后进行传输,前端收到后需要反编码才能获取到真正的名称
            response.addHeader("Content-Disposition", "attachment;filename=" + URLEncoder.encode(filename, "UTF-8"));
            // 告知浏览器文件的大小
            response.addHeader("Content-Length", "" + file.length());
            OutputStream outputStream = new BufferedOutputStream(response.getOutputStream());
            response.setContentType("application/octet-stream");
            outputStream.write(buffer);
            outputStream.flush();
            return "111";
        } catch (IOException ex) {
            log.error(ex.toString());
        }
        return "222";
    }

    @RequestMapping("/downloadPdf")
    public void downloadPdf(HttpServletResponse response) throws IOException, DocumentException {
        //第一步
        // 清空response
        response.reset();
        // 设置response的Header
        response.setCharacterEncoding("UTF-8");
        //Content-Disposition的作用：告知浏览器以何种方式显示响应返回的文件，用浏览器打开还是以附件的形式下载到本地保存
        //attachment表示以附件方式下载   inline表示在线打开   "Content-Disposition: inline; filename=文件名.mp3"
        // filename表示文件的默认名称，因为网络传输只支持URL编码的相关支付，因此需要将文件名URL编码后进行传输,前端收到后需要反编码才能获取到真正的名称
        response.addHeader("Content-Disposition", "attachment;filename=" + URLEncoder.encode("testoone3.pdf", "UTF-8"));

        // 第二步
        OutputStream os = response.getOutputStream();
        //设置读取html的文件路径 如果使用字符串setDocumentFromString
        ITextRenderer renderer = new ITextRenderer();
        renderer.setDocument("file:/D:/aaa/aaa.html");
        //设置字体 以jar包形式运行的话 addFont只能输入路径 要以流的方式 所以需要临时文件
        //创建临时字体文件
        InputStream fontStream = ResourceUtil.getStream("fonts/simsun.ttc");
        File tempFontFile = File.createTempFile("simsun", ".ttc");
        FileUtil.writeFromStream(fontStream, tempFontFile);
        //将临时字体文件添加
        ITextFontResolver fontResolver = renderer.getFontResolver();
        fontResolver.addFont(tempFontFile.getAbsolutePath(), BaseFont.IDENTITY_H, BaseFont.NOT_EMBEDDED);
        //生成pdf
        renderer.layout();
        renderer.createPDF(os);
        os.close();
    }
}
